// Protocol Buffers for GWT
// Copyright 2010 Thomas Broyer  All rights reserved.
// http://code.google.com/p/protobuf-gwt/
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
//     * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above
// copyright notice, this list of conditions and the following disclaimer
// in the documentation and/or other materials provided with the
// distribution.
//     * Neither the name of Google Inc. nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

package net.ltgt.gwt.protobuf.compiler.client;

import com.google.protobuf.Descriptors.FieldDescriptor;
import com.google.protobuf.compiler.JavaHelpers;

import java.io.IOException;
import java.io.Writer;

/**
 * @author t.broyer@ltgt.net Thomas Broyer
 * Based on the initial work of:
 * @author kenton@google.com Kenton Varda
 * Based on original Protocol Buffers design by
 * Sanjay Ghemawat, Jeff Dean, and others.
 */
public class RepeatedEnumFieldGenerator extends RepeatedFieldGenerator {

  protected RepeatedEnumFieldGenerator(FieldDescriptor descriptor) {
    super(descriptor, JavaHelpers.getClassName(descriptor.getEnumType()));
  }
  @Override
  public void generateBuilderMembers(Writer output) throws IOException {
    output.write("public java.util.List<" + typeName + "> get" + capitalizedName + "List() {\n");
    output.write("  return result.get" + capitalizedName + "List();\n");
    output.write("}\n");
    output.write("public int get" + capitalizedName + "Count() { return result.get" + capitalizedName + "Count(); }\n");
    output.write("public " + typeName + " get" + capitalizedName + "(int index) {\n");
    output.write("  return result.get" + capitalizedName + "(index);\n");
    output.write("}\n");
    output.write("public Builder set" + capitalizedName + "(int index, " + typeName + " value) {\n");
    output.write("  if (value == null) { throw new NullPointerException(); }\n");
    output.write("  result." + name + "_.set(index, value);\n");
    output.write("  return this;\n");
    output.write("}\n");
    output.write("public Builder add" + capitalizedName + "(" + typeName + " value) {\n");
    output.write("  if (value == null) { throw new NullPointerException(); }\n");
    output.write("  if (result." + name + "_ == null) { result." + name + "_ = new java.util.ArrayList<" + typeName + ">(); }\n");
    output.write("  result." + name + "_.add(value);\n");
    output.write("  return this;\n");
    output.write("}\n");
    output.write("public Builder addAll" + capitalizedName + "(java.lang.Iterable<? extends " + typeName + "> values) {\n");
    output.write("  if (values == null) { throw new NullPointerException(); }\n");
    output.write("  if (result." + name + "_ == null) { result." + name + "_ = new java.util.ArrayList<" + typeName + ">(); }\n");
    output.write("  super.addAll(values, result." + name + "_);\n");
    output.write("  return this;\n");
    output.write("}\n");
    output.write("public Builder clear" + capitalizedName + "() {\n");
    output.write("  result." + name + "_ = null;\n");
    output.write("  return this;\n");
    output.write("}\n");
  }

  @Override
  public void generateParsingCode(Writer output) throws IOException {
    // TODO Auto-generated method stub

  }
  @Override
  public void generateSerializationCode(Writer output) throws IOException {
    // TODO Auto-generated method stub
    
  }

  @Override
  public void generateDescriptorCode(Writer output) throws IOException {
    output.write("new com.google.protobuf.Descriptors.RepeatedEnumFieldDescriptor(this,\n");
    output.write("    \"" + descriptor.getName() + "\", " + descriptor.getNumber() + ", " + descriptor.getIndex() + ") {\n");
    output.write("  public com.google.protobuf.Descriptors.EnumDescriptor getEnumType() {\n");
    output.write("    return " + JavaHelpers.getClassName(descriptor.getEnumType()) + ".getDescriptor();\n");
    output.write("  }\n");
    output.write("},\n");
  }
  
  @Override
  public void generateAddRepeatedFieldCode(Writer output) throws IOException {
    output.write("case " + descriptor.getNumber() + ":\n");
    output.write("  add" + capitalizedName + "(" + boxedTypeName + ".valueOf(\n");
    output.write("    ((com.google.protobuf.Descriptors.EnumValueDescriptor) value).getNumber()));\n");
    output.write("  break;\n");
  }

  @Override
  public void generateGetFieldCode(Writer output) throws IOException {
    output.write("case " + descriptor.getNumber() + ": {\n");
    output.write("  java.util.List<com.google.protobuf.Descriptors.EnumValueDescriptor> ret =\n");
    output.write("    new java.util.ArrayList<com.google.protobuf.Descriptors.EnumValueDescriptor>();\n");
    output.write("  for (" + boxedTypeName + " element : get" + capitalizedName + "List()) {\n");
    output.write("    ret.add(element.getValueDescriptor());\n");
    output.write("  }\n");
    output.write("  return java.util.Collections.unmodifiableList(ret);\n");
    output.write("}\n");
  }
  
  @Override
  public void generateGetRepeatedFieldCode(Writer output) throws IOException {
    output.write("case " + descriptor.getNumber() + ":\n");
    output.write("  return get" + capitalizedName + "(index).getValueDescriptor();\n");
  }
  
  @Override
  public void generateSetFieldCode(Writer output) throws IOException {
    output.write("case " + descriptor.getNumber() + ":\n");
    output.write("  clear" + capitalizedName + "();\n");
    output.write("  for (final com.google.protobuf.Descriptors.EnumValueDescriptor element\n");
    output.write("      : (java.util.List<com.google.protobuf.Descriptors.EnumValueDescriptor>) value) {\n");
    output.write("    add" + capitalizedName + "(" + boxedTypeName + ".valueOf(element.getNumber()));\n");
    output.write("  }\n");
    output.write("  break;");
  }
  
  @Override
  public void generateSetRepeatedFieldCode(Writer output) throws IOException {
    output.write("case " + descriptor.getNumber() + ":\n");
    output.write("  set" + capitalizedName + "(index, " + boxedTypeName + ".valueOf(\n");
    output.write("    ((com.google.protobuf.Descriptors.EnumValueDescriptor) value).getNumber()));\n");
    output.write("  break;\n");
  }
}
